package org.dromara.common.core.utils;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.lang.tree.Tree;
import cn.hutool.core.lang.tree.TreeNodeConfig;
import cn.hutool.core.lang.tree.TreeUtil;
import cn.hutool.core.lang.tree.parser.NodeParser;
import org.dromara.common.core.utils.reflect.ReflectUtils;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;

import java.util.List;


/**
 * @Author 刘武贵
 * @Date 2024/8/15 22:00
 * @Description 功能描述：扩展 hutool TreeUtil 封装系统树构建
 */
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class TreeBuildUtils extends TreeUtil {

    /**
     * 根据前端定制差异化字段
     */
    public static final TreeNodeConfig DEFAULT_CONFIG = TreeNodeConfig.DEFAULT_CONFIG.setNameKey("label");

    /**
     * 构建树形结构
     *
     * @param list 待构建树形结构的数据列表
     * @param nodeParser 用于解析节点数据的函数式接口
     * @param <T> 泛型参数，表示列表中元素的类型
     * @param <K> 泛型参数，表示树节点标识的类型
     * @return 返回构建好的树形结构列表，如果输入列表为空，则返回null
     */
    public static <T, K> List<Tree<K>> build(List<T> list, NodeParser<T, K> nodeParser) {
        // 检查列表是否为空，如果为空则返回null，避免后续操作
        if (CollUtil.isEmpty(list)) {
            return null;
        }
        // 获取第一个元素的parentId作为根节点标识，用于构建树形结构
        K k = ReflectUtils.invokeGetter(list.get(0), "parentId");
        // 调用TreeUtil.build方法构建树形结构，并返回结果
        return TreeUtil.build(list, k, DEFAULT_CONFIG, nodeParser);
    }

}
